package com.qf.listener.simple;

import com.rabbitmq.client.Channel;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.annotation.RabbitHandler;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;

import java.io.IOException;
import java.util.concurrent.TimeUnit;

/**
 * @RabbitListener: 监听器 器 注解，用来监听对列中是否有消息
 * @RabbitHandler； 执行处理消息的方法
 * @author 千锋教育
 * @Company http://www.mobiletrain.org/
 * @Version 1.0
 */
@Component
@RabbitListener(queues = {"simple_queue"})
public class SimpleMessageListener {

    @Autowired
    StringRedisTemplate redisTemplate;

    @RabbitHandler
    public void consum(String msg, Channel channel, Message message) throws IOException {

        String messageId = (String) message.getMessageProperties().getHeaders().get("spring_returned_message_correlation");

        //setIfAbsent 相当于 redis中的 setnx 0 ,1
        boolean flag = redisTemplate.opsForValue().setIfAbsent(messageId, "0", 10, TimeUnit.MINUTES);
        //flag 为true，没有id， 正常消费
        if(flag){
            try {
                System.out.println("消息队列中的消息:" + msg);
//            int i = 1/0;
                //执行成功
                /**
                 * 参数1： 队列消息的唯一的标识
                 * 参数2：是否批量处理
                 */
                channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
                redisTemplate.delete(messageId);
            } catch (Exception e) {
                e.printStackTrace();
                // 出现异常
                channel.basicNack(message.getMessageProperties().getDeliveryTag(), false, true);

            }
        }
        //flag 为false， 说明redis已经存在id，
        //判断 messgeId对应的value是0， 还是1  ，如果0，正在消费，如果为1， 消费成功，手动ack
        else {
            String status = redisTemplate.opsForValue().get(messageId);
            if("1".equalsIgnoreCase(status)){
                channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
                redisTemplate.delete(messageId);
            }

        }




    }
}
